from django_redis import get_redis_connection


class RedisTool:
    def __init__(self):
        self.redis = get_redis_connection('code')

    # 封装一个存储验证码的函数 有效期2分钟
    def set_code(self, phone, code):  # key  val   expire
        # setex key键 expire有效期 val值
        self.redis.setex(phone, 600, code)

    def get_code(self, phone):
        code = self.redis.get(phone)
        if not code:
            return None
        return code.decode('utf-8')  # 反回字符串 将二进制转换为utf-8


# 单例模型
redistool = RedisTool()
